import numpy as np

# 一维数组的分割
print("一维数组的分割".center(100, '='))
x1 = np.arange(1, 9)
print(x1)
x2 = np.split(x1, 2)
print(x2)
print(x2[0])
x3 = np.split(x1, [2, 6])  # [2, 6]表示以2为左边界，6为右边界分割 ， 2包含在左边数组中， 6在分割中间数组中
print(x3)

# 二维数组的分割
print("二维数组的分割".center(100, '='))
a = np.arange(1, 13).reshape(3, 4)
print(a)
print('axis = 0 , 将二维数组a进行垂直分割')
b = np.split(a, 3, axis=0)  # 3表示分割个数
print(b)
print('axis = 1 , 将二维数组a进行水平分割')
b = np.split(a, 2, axis=1)  # 3表示分割个数
print(b)
print('使用hsplit(,)函数将二维数组进行水平分割')
b = np.hsplit(a, 4)
print(b)
print('使用vsplit(,)函数将二维数组进行垂直分割')
b = np.vsplit(a, 3)
print(b)
